tensor([[0.7200, 0.7100, 0.7200, 0.7100, 0.7200],
        [0.2800, 0.2900, 0.2800, 0.2900, 0.2800]], grad_fn=<RoundBackward1>)